/*
 * This file is part of cve-check-tool
 * Copyright (C) 2015 Intel Corporation
 *
 * cve-check-tool is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 */

#define _GNU_SOURCE
#include <check.h>
#include <stdio.h>
#include <stdlib.h>

#include "cve-string.c"
#include "hashmap.c"
#include "util.c"
#include "util.h"

#include "config.h"

const char *nvd_file = "nvd.db";

/**
 * Ensure parse_xml_date works
 */
START_TEST(cve_date_function)
{
        const char *date = "2015-03-10T08:24:10.220-05:00";
        int64_t t1, t2;

        t1 = parse_xml_date(date);
        fail_if(t1 < 0, "Failed to parse XML date");

        date = "2015-03-10T08:34:10.220-05:00";
        t2 = parse_xml_date(date);
        fail_if(t2 < 0, "Failed to parse second XML date");

        fail_if(t1 > t2, "XML Date 1 should be less than Date 2");
}
END_TEST

/**
 * Ensure scope based management is functional
 */
static bool reclaimed = false;
typedef char memtestchar;

static void reaper(void *v)
{
        free(v);
        v = NULL;
        fprintf(stderr, "Freeing tmp var\n");
        reclaimed = true;
}
DEF_AUTOFREE(memtestchar, reaper)

START_TEST(cve_memory_test)
{
        {
                autofree(memtestchar) *tmp = NULL;

                if (!asprintf(&tmp, "Allocation test")) {
                        fail("Unable to allocate memory");
                }
        }
        fail_if(!reclaimed, "Scope based tmp var was not reclaimed!");
}
END_TEST

START_TEST(cve_string_test)
{
        cve_string *str = NULL, *str2 = NULL;
        char *c = NULL;

        str = cve_string_dup_printf("Test String #%d", 1);
        fail_if(!str, "Failed to allocate string");
        fail_if(!g_str_equal(str->str, "Test String #1"), "Invalid formatted string");
        fail_if(cstrlen(str) != 14, "Incorrect string length");

        fail_if(!cve_string_has_suffix(str, "g #1"), "String has incorrect suffix");
        fail_if(!cve_string_has_prefix(str, "T"), "String has incorrect prefix");

        fail_if(!cve_string_cat(str, "append"), "Failed to append string");
        fail_if(cstrlen(str) != 20, "Incorrect string length after append");

        str2 = cve_string_dup(str->str);
        fail_if(!g_str_equal(str->str, str2->str), "Strings do not match");
        fail_if(cstrlen(str) != cstrlen(str2), "Invalid string length after dup");

        cve_string_free(str);
        cve_string_free(str2);

        str = cve_string_dup("  spacey text  ");
        c = cve_string_strip(str);
        fail_if(!c, "Failed to strip string");
        fail_if(!streq(c, "spacey text"), "Failed to match stripped string");

        str = cve_string_dup_printf("Test String #%d", 2);
        fail_if(!str, "Failed to allocate string");
        fail_if(!cve_string_const_equal(str, "Test String #2"), "Const String Compare failed");
        /* Ensure its not failing completely. */
        fail_if(cve_string_const_equal(str, "Test String #1"), "Const String Compare should not match");

        str2 = cve_string_dup(str->str);
        fail_if(!str2, "Failed to dup string");
        fail_if(!cve_string_equal(str, str2), "Identical strings not matching");

        fail_if(!cve_string_const_equal(str2, (const char *)str->str), "Direct const compare fail!");

        cve_string_free(str);
        cve_string_free(str2);

        fail_if(cve_string_dup(NULL), "dup on NULL string");
        fail_if(cve_string_dup_printf(NULL), "dup_printf on NULL string");
        fail_if(cve_string_cat(NULL, NULL), "cat on NULL string");

        fail_if(cve_string_equal(NULL, NULL), "equal on NULL string");
        fail_if(cve_string_const_equal(NULL, NULL), "const_equal on NULL string");

        /* Forced empty ->str tests */
        cve_string st = {.len = 0 };
        fail_if(cve_string_equal(&st, &st), "equal on NULL ->str");
        fail_if(cve_string_const_equal(&st, "TEST"), "const_equal on NULL ->str");
}
END_TEST

static Suite *core_suite(void)
{
        Suite *s = NULL;
        TCase *tc = NULL;

        s = suite_create("cve_core");
        tc = tcase_create("cve_core_functions");
        tcase_add_test(tc, cve_date_function);
        tcase_add_test(tc, cve_memory_test);
        tcase_add_test(tc, cve_string_test);
        suite_add_tcase(s, tc);

        return s;
}

int main(void)
{
        Suite *s;
        SRunner *sr;
        int fail;

        s = core_suite();
        sr = srunner_create(s);
        srunner_run_all(sr, CK_VERBOSE);
        fail = srunner_ntests_failed(sr);
        srunner_free(sr);

        if (fail > 0) {
                return EXIT_FAILURE;
        }

        return EXIT_SUCCESS;
}

/*
 * Editor modelines  -  https://www.wireshark.org/tools/modelines.html
 *
 * Local variables:
 * c-basic-offset: 8
 * tab-width: 8
 * indent-tabs-mode: nil
 * End:
 *
 * vi: set shiftwidth=8 tabstop=8 expandtab:
 * :indentSize=8:tabSize=8:noTabs=true:
 */
